import zipfile
import os

# 定义压缩文件所在文件夹路径和目标解压文件夹路径
zip_folder = './data/CelebA/Img'  # 压缩文件所在文件夹
extract_folder = './data/CelebA/image'  # 解压到的目标文件夹

# 创建解压文件夹（如果不存在）
os.makedirs(extract_folder, exist_ok=True)

# 遍历文件夹中的所有文件
for filename in os.listdir(zip_folder):
    # 检查文件是否为 .zip 文件
    if filename.endswith('.zip'):
        file_path = os.path.join(zip_folder, filename)
        print(f"解压文件: {file_path}")

        # 解压缩文件到指定文件夹
        with zipfile.ZipFile(file_path, 'r') as zip_ref:
            zip_ref.extractall(extract_folder)
        print(f"{filename} 解压完成！")

print("所有文件已解压到 image 目录。")
